leetcode 808 problem solution in python